import numpy as np

arrive = np.array([1, 1])
tai_bei = np.array([1, 1])
on = np.array([2, 2])

weight = np.array([[1, 1], [1, 1]])
h_0 = np.array([0, 0])

h_1 = h_0 @ weight + arrive @ weight
print('h_1:', h_1)
o_1 = h_1 @ weight
print('o_1:', o_1)

h_2 = h_1 @ weight + tai_bei @ weight
print('h_2:', h_2)
o_2 = h_2 @ weight
print('o_2:', o_2)


h_3 = h_2 @ weight + on @ weight
print('h_3:', h_3)
o_3 = h_3 @ weight
print('o_3:', o_3)
